#include <QCoreApplication>
#include <QDebug>
#define __STDC_CONSTANT_MACORS

extern "C"
{
#include "libavformat/avformat.h"
}

int close();

AVOutputFormat *ofmt = NULL;

//输入输出各对应一个AVFormatContext
AVFormatContext *ifmt_ctx = NULL, *ofmt_ctx = NULL;
int ret;

int main(int argc,char* argv[]){
    QCoreApplication a(argc,argv);

    AVPacket pkt;
    const char *in_filename, *out_filename;
    int i;

    in_filename  = "D:/1.flv";      //输入文件名（Input file URL）
    out_filename = "D:/2.flv";      //输出文件名（Output file URL）

    qDebug() << "BEGIN" << endl;
    av_register_all();

    // 打开输入文件
    if ((ret = avformat_open_input(&ifmt_ctx, in_filename, 0, 0)) < 0) {
        qDebug() <<  "Could not open input file." << endl;
        return close();
    }
    // 获取媒体信息
    if((ret = avformat_find_stream_info(ifmt_ctx,0))<0){
        qDebug() <<  "Failed to retrieve input stream information" << endl;
        return close();
    }
    // 打印输入格式的详细信息，比如持续时间duration,比特率 bitrate,流 streams,容器 container, 程序programs,元数据 metadata, side data, 编解码　　 器codec ，时基 time base.
    // 第2个参数，流索引
    // 第3个参数，打印的url，如源文件或目标文件
    // 第4个参数，is_output 选择指定的context是0输入 还是1输出
    av_dump_format(ifmt_ctx, 0, in_filename, 0);

    // 分配一个用于输出格式的AVFormatContext
    avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);
    if(!ofmt_ctx){
        qDebug() <<  "Could not create output context\n" << endl;
        ret = AVERROR_UNKNOWN;
        return close();
    }
    qDebug() <<  "hello" << endl;
    ofmt = ofmt_ctx->oformat;

    for(i=0;i<ifmt_ctx->nb_streams;i++){
        //根据输入流创建输出流（Create output AVStream according to input AVStream）
        AVStream *in_stream = ifmt_ctx->streams[i];
        AVStream *out_stream = avformat_new_stream(ofmt_ctx, in_stream->codec->codec);
        if (!out_stream) {
            printf( "Failed allocating output stream\n");
            ret = AVERROR_UNKNOWN;
            return close();
        }
        // 拷贝源AVCodecContext的设置到目的AVCodecContext
        ret = avcodec_copy_context(out_stream->codec, in_stream->codec);
        if (ret < 0) {
            printf( "Failed to copy context from input to output stream codec context\n");
            return close();
        }
        out_stream->codec->codec_tag = 0;
        if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
            out_stream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; //CODEC_FLAG_GLOBAL_HEADER找不到，使用AV_CODEC_FLAG_GLOBAL_HEADER;
    }
    //打印输出格式详细信息
    av_dump_format(ofmt_ctx, 0, out_filename, 1);

    //打开输出文件（Open output file）
     if (!(ofmt->flags & AVFMT_NOFILE)) {
         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE);
         if (ret < 0) {
             printf( "Could not open output file '%s'", out_filename);
             return close();
         }
     }
     //写文件头（Write file header）
     ret = avformat_write_header(ofmt_ctx, NULL);
     if (ret < 0) {
         printf( "Error occurred when opening output file\n");
         return close();
     }
     int frame_index=0;

     while (1) {
             AVStream *in_stream, *out_stream;
             //获取一个AVPacket（Get an AVPacket）
             ret = av_read_frame(ifmt_ctx, &pkt);
             if (ret < 0)
                 break;
             in_stream  = ifmt_ctx->streams[pkt.stream_index];
             out_stream = ofmt_ctx->streams[pkt.stream_index];
             /* copy packet */
             //转换PTS/DTS（Convert PTS/DTS）

             pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));
             pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));
             pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);
             pkt.pos = -1;
             //写入（Write）
             ret = av_interleaved_write_frame(ofmt_ctx, &pkt);
             if (ret < 0) {
                 printf( "Error muxing packet\n");
                 break;
             }
             //printf("Write %8d frames to output file\n",frame_index);
             av_free_packet(&pkt);
             frame_index++;
         }
         //写文件尾（Write file trailer）
         av_write_trailer(ofmt_ctx);

    return a.exec();
}
// 释放资源
int close(){
    avformat_close_input(&ifmt_ctx);
    /* close output */
    if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE))
        avio_close(ofmt_ctx->pb);
    avformat_free_context(ofmt_ctx);
    if (ret < 0 && ret != AVERROR_EOF) {
        printf( "Error occurred.\n");
        return -1;
    }
    return 0;
}
